# -*- coding: utf-8 -*-
"""
@Time    : 2025/3/17 14:06 
@Author  : ZhangShenao 
@File    : 1. 文生图.py
@Desc    : 使用OpenAI的dall-e-3模型,生成图片
"""
import os

import dotenv
# 导入openai依赖库
from openai import OpenAI

# 加载环境变量
dotenv.load_dotenv()

# 创建OpenAI客户端
client = OpenAI(base_url=os.getenv("OPENAI_API_BASE"))

# 调用images.generate接口,生成图片
resp = client.images.generate(
    model="dall-e-3",  # 指定模型,可选dall-e-2和dall-e-3(默认)
    prompt="一只英短蓝猫，正躺在沙发上睡觉，非常惬意",  # 生成图片的Prompt
    n=1,  # 生成图片的数量
    quality="hd",  # 图片质量,可选hd和standard(默认)
    size="1024x1024",  # 图片尺寸
    style="natural",  # 图片风格,可选vivid和natural(默认)
    response_format="url",  # 返回图片的格式,可选url和b64_json(默认)
)

# 打印生成图片的url
print(resp.data[0].url)
